plt.figure(figsize=(20, 10))
f = plt.subplot(121)
f.set_title('input image', fontsize=15)
plt.imshow(img)
out = y.numpy()#(1, 280, 376, 1)
f = plt.subplot(122)
f.set_title('output feature map', fontsize=15)
#通过numpy.squeeze函数将大小为1的维度消除
plt.imshow(out.squeeze(), cmap='gray')
plt.show() 
